conda create -n rocl python=3.6
conda activate rocl

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
conda install opencv
conda install scipy

pip install torchlars
pip install git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
pip install diffdist